#!/bin/bash

# IK分词器安装脚本 - 优化版本
echo "开始安装IK分词器..."

# 设置环境变量
IK_VERSION="8.4.1"
IK_URL="https://get.infini.cloud/elasticsearch/analysis-ik/${IK_VERSION}"
PLUGIN_DIR="/usr/share/elasticsearch/plugins/analysis-ik"

# 检查插件是否已安装
if [ -d "$PLUGIN_DIR" ] && [ -f "$PLUGIN_DIR/plugin-descriptor.properties" ]; then
    echo "IK分词器已安装，跳过安装步骤"
    exit 0
fi

echo "正在安装IK分词器..."

# 确保有下载工具
if ! command -v curl &> /dev/null; then
    echo "安装curl..."
    apt-get update -qq && apt-get install -y curl
fi

# 创建插件目录
mkdir -p "$PLUGIN_DIR"
cd /tmp

# 下载IK插件
echo "正在下载IK分词器..."
if curl -L -f -o "analysis-ik-${IK_VERSION}.zip" "$IK_URL"; then
    echo "下载成功，开始解压..."
    
    # 解压到插件目录
    if unzip -q "analysis-ik-${IK_VERSION}.zip" -d "$PLUGIN_DIR"; then
        echo "IK分词器安装成功！"
        
        # 设置权限 - 简化权限设置
        echo "设置插件权限..."
        
        # 清理临时文件
        rm -f "analysis-ik-${IK_VERSION}.zip"
        
        echo "权限设置完成"
    else
        echo "解压失败"
        rm -rf "$PLUGIN_DIR"
        exit 1
    fi
else
    echo "下载失败，尝试备用地址..."
    # 可以在这里添加备用下载地址
    echo "安装失败，请检查网络连接"
    exit 1
fi

echo "IK分词器安装完成"